package ai.zixing.example03;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 事件处理类
 */
public class EventManager {

    // 监听器集合
    Map<Enum<EventType>, List<EventListener>> listenerMap = new HashMap<>();

    // 构造器，提高扩展性
    public EventManager(Enum<EventType>... operations) {
        for (Enum<EventType> operation : operations) {
            this.listenerMap.put(operation, new ArrayList<>());
        }
    }

    /**
     * 订阅
     *
     * @param typeEnum 事件类型
     * @param listener 监听对象
     */
    public void subscribe(Enum<EventType> typeEnum, EventListener listener) {
        List<EventListener> users = listenerMap.get(typeEnum);
        users.add(listener);
    }

    /**
     * 取消订阅
     *
     * @param typeEnum 事件类型
     * @param listener 监听对象
     */
    public void unSubscribe(Enum<EventType> typeEnum, EventListener listener) {
        List<EventListener> users = listenerMap.get(typeEnum);
        users.remove(listener);
    }

    /**
     * 通知
     *
     * @param eventType 事件类型
     * @param result    结果
     */
    public void notify(Enum<EventType> eventType, LotteryResult result){
        List<EventListener> users = listenerMap.get(eventType);
        for (EventListener listener : users) {
            listener.doEvent(result);
        }
    }
}
